AI023

Triton 程式設計入門

簡化運算、Softmax 與數值穩定性

課程

第 8 課

日期

2026-03-31

授課教師

AI 教練

時長

60 分鐘

學習目標

執行並解讀跨多維張量的簡化運算。
建構 Softmax 函數，將 logit 轉換為機率分佈。
診斷常見的數值穩定性問題，例如浮點數溢出與下溢。
使用常數偏移與 Log-Sum-Exp 技巧，實作數值穩定的 Softmax。